package com.team.ai.service;

import com.team.ai.tool.Juhe.Current.RequestTool;
import net.sf.json.JSONObject;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;
@Component
public class JuHeService {

    private String errMsg="不好意思，暂时无法提供服务";//查询出错的提示
    private String nullMsg="查询不到您想要的数据";//查询为空的提示

    /**
     * chatgpt
     * @param q 用户输入的问题
     * @return
     */

    public String chatgpt(String q){
        String url="http://gpt.juhe.cn/chatgpt/query";//请求URL
        String key="a25917cd8ed285086da55ef2ffe1a9c2";//请求的Key
        String res="";//请求的结果
        Map<String,Object> params= new HashMap<>();//传入的参数
        params.put("key",key);
        params.put("q",q);
        RequestTool requestTool = new RequestTool();
        String queryParam= requestTool.urlencode(params);//拼接Get请求参数部分
        String response=requestTool.doGet(url,queryParam);//获取返回结果
        try {
            JSONObject jsonObject = JSONObject.fromObject(response);//转成JSON对象
            int errCode= jsonObject.getInt("error_code");//取出错误代码
            if(errCode==0){//请求成功！
                JSONObject result = jsonObject.getJSONObject("result");//将JSON中的result对象取出
                res =result.getString("resp");
            }else {//请求失败！
                res= nullMsg;
            }
        }catch (Exception e){
            e.printStackTrace();
            res=errMsg;
        }
        return  res;
    }
}
